package com.zboin.interview150;

import java.util.*;

/** Hard
 * 给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。
 *
 *  s 中的 串联子串 是指一个包含  words 中所有字符串以任意顺序排列连接起来的子串。
 *
 * 例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab" 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。
 * 返回所有串联子串在 s 中的开始索引。你可以以 任意顺序 返回答案。
 *
 *
 *
 * 示例 1：
 *
 * 输入：s = "barfoothefoobarman", words = ["foo","bar"]
 * 输出：[0,9]
 * 解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
 * 子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
 * 子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
 * 输出顺序无关紧要。返回 [9,0] 也是可以的。
 * 示例 2：
 *
 * 输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
 * 输出：[]
 * 解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
 * s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
 * 所以我们返回一个空数组。
 * 示例 3：
 *
 * 输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
 * 输出：[6,9,12]
 * 解释：因为 words.length == 3 并且 words[i].length == 3，所以串联子串的长度必须为 9。
 * 子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
 * 子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
 * 子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。
 *
 *
 * 提示：
 *
 * 1 <= s.length <= 104
 * 1 <= words.length <= 5000
 * 1 <= words[i].length <= 30
 * words[i] 和 s 由小写英文字母组成
 */
public class Course30 {

    public static void main(String[] args) {
        List<Result> list = List.of(
                new Result("barfoothefoobarman", new String[]{"foo","bar"}),
                new Result("wordgoodgoodgoodbestword", new String[]{"word","good","best","word"}),
                new Result("barfoofoobarthefoobarman", new String[]{"bar","foo","the"}),
                new Result("wordgoodgoodgoodbestword", new String[]{"word","good","best","good"})
        );

        Course30 c = new Course30();

        list.forEach(item -> {
            List<Integer> result = c.findSubstring(item.str, item.words);
            result.forEach(i -> System.out.print(i + ", "));
            System.out.println();
        });
    }

    public List<Integer> findSubstring(String s, String[] words) {
        int row = words.length, col = words[0].length();
        int n = row * col;
        int len = s.length();
        List<Integer> res = new ArrayList<>();

        if (s.length() < n) return res;
        HashMap<String, Integer> targetCount = new HashMap<>(row);
        for (String word : words) {
            targetCount.put(word, targetCount.getOrDefault(word, 0) + 1);
        }

        for (int offset = 0; offset < col; offset++) {
            int left = offset, right = offset, curMatch = 0;
            Map<String, Integer> currentWindowMap = new HashMap<>(row);

            while (right + col <= len) {
                // 提取右侧的单词
                String str = s.substring(right, right + col);
                right += col;

                // 如果当前单词为 Target 中不存在的内容，则直接跳到下一个单词重新开始
                if (!targetCount.containsKey(str)) {
                    left = right;
                    currentWindowMap.clear();
                    curMatch = 0;
                    continue;
                }

                currentWindowMap.put(str, currentWindowMap.getOrDefault(str, 0) + 1);
                curMatch++;

                // 如果单词数大于 target，则从左侧窗口收缩，直到目标 target 数量匹配
                while (currentWindowMap.get(str) > targetCount.get(str)) {
                    String leftWord = s.substring(left, left + col);
                    currentWindowMap.put(leftWord, currentWindowMap.getOrDefault(leftWord, 0) - 1);
                    left += col;
                    curMatch--;
                }

                if (curMatch == row) {
                    res.add(left);

                    // 继续滑动窗口， left 指针收缩一个单词
                    String leftWord = s.substring(left, left + col);
                    currentWindowMap.put(leftWord, currentWindowMap.getOrDefault(leftWord, 0) - 1);
                    left += col;
                    curMatch--;
                }
            }
        }

        return res;
    }

    /**
     * 暴力解法，超时
     */
    public List<Integer> findSubstring_force(String s, String[] words) {
        int r = words.length, c = words[0].length();
        int n = r * c;
        List<Integer> res = new ArrayList<>();

        if (s.length() < n) return res;
        HashMap<String, Integer> targetCount = new HashMap<>(r);
        for (String word : words) {
            targetCount.put(word, targetCount.getOrDefault(word, 0) + 1);
        }

        for (int i = 0; i <= s.length() - n; i++) {

            // 提取接下来 r * c 长度的字串，并提取 r 个 c 长度的单词
            int p = i;
            HashMap<String, Integer> wordCount = new HashMap<>(r);
            boolean found = true;
            for (int j = 0; j < r; j++) {
                String str = s.substring(p, p + c);
                if (!targetCount.containsKey(str)) {
                    found = false;
                    break;
                }
                wordCount.put(str, targetCount.getOrDefault(str, 0) + 1);
                p += c;
            }
            if (!found) continue;

            // 比对两个 map 中key以及count是否一致
            for (Map.Entry<String, Integer> entry : targetCount.entrySet()) {
                Integer val = wordCount.get(entry.getKey());
                if (val == null || !val.equals(entry.getValue())) {
                    found = false;
                    break;
                }
            }
            if (found) {
                res.add(i);
            }
        }

        return res;
    }

    private record Result(String str, String[] words) { }
}
